# 首页面对象
from selenium.webdriver.common.by import By

from utils import UtilsDriver


class HomePage:    # 对象库层
    # 登录跳转按钮
    def __init__(self):
        self.driver = UtilsDriver.get_driver()  # 通过utils获取驱动对象

    # 定义获取登录按钮的元素对象
    def get_go_login_btn(self):
        return self.driver.find_element(By.CLASS_NAME, "red")


class HomeHandle:  # 操作层
    def __init__(self):
        self.home_page = HomePage()  # 实例化对象库层的类

    # 定义登录按钮的操作方法
    def click_go_login(self):
        self.home_page.get_go_login_btn().click()


class HomeProxy:  # 业务层
    def __init__(self):
        self.home_handle = HomeHandle()  # 实例化操作层的类

    # 定义业务操作
    # 跳转到登录页面
    def go_to_login_page(self):
        self.home_handle.click_go_login()
